import os
import shutil
from pathlib import Path

def cleanup():
    # 获取项目根目录
    rootDir = Path(__file__).parent.parent.absolute()
    
    # 要清理的目录
    tmpDir = rootDir / "tmp"
    distDir = rootDir / "dist"
    srcDir = rootDir / "src"
    
    # 清理tmp目录
    if tmpDir.exists():
        print(f"清理 {tmpDir}...")
        shutil.rmtree(tmpDir)
        tmpDir.mkdir(exist_ok=True)
    
    # 清理dist目录
    if distDir.exists():
        print(f"清理 {distDir}...")
        shutil.rmtree(distDir)
    
    # 清理src目录下的所有__pycache__目录
    for pycache in srcDir.rglob("__pycache__"):
        print(f"清理 {pycache}...")
        shutil.rmtree(pycache)
    
    print("清理完成！")

if __name__ == "__main__":
    cleanup() 